# 员工管理模块，封装接口对象层
import requests
class EmpManageApi(object):
    # 定义员工管理类
    def add_emp(self,req_header,req_body):
        # 添加员工
        resp = requests.post(url="http://ihrm2-test.itheima.net/api/sys/user",
                             headers=req_header,
                             json=req_body)
        return resp
    def query_emp(self,emp_id,req_header):
        # 查询员工
        resp = requests.get(url=" http://ihrm2-test.itheima.net/api/sys/user/"+emp_id,
                            headers=req_header)
        return resp

    def modify_emp(self,emp_id,req_header,req_body):
        # 修改员工
        resp = requests.put(url=" http://ihrm2-test.itheima.net/api/sys/user/" + emp_id,
                            headers=req_header,
                            json=req_body)
        return resp
    def del_emp(self,emp_id,req_header):
        # 删除员工
        resp = requests.delete(url=" http://ihrm2-test.itheima.net/api/sys/user/" + emp_id,
                            headers=req_header)
        return resp
if __name__ == '__main__':
    emp=EmpManageApi()
    # 准备请求头
    resp = requests.post(url=" http://ihrm2-test.itheima.net/api/sys/login",
                         json={"mobile": "13800000002", "password": "123456"})
    token = resp.json().get("data")
    #
    req_header={"Content-Type":"application/json",
                       "Authorization":token}
    # 准备请求体
    req_body ={
                  "username":"小傻瓜001",
                  "mobile":"13922159112",
                  "workNumber":"9588"
              }
    resp=emp.add_emp(req_header,req_body)
    print(resp.json())

    # 测试查询员工
    emp_id = "1549411070379044864"
    resp=emp.query_emp(emp_id,req_header)
    print(resp.json())

    # 测试修改员工
    req_body = {"username":"小傻猪猪"}
    resp=emp.modify_emp(emp_id,req_header,req_body)
    print(resp.json())

    # 测试删除员工
    resp=emp.del_emp(emp_id,req_header)
    print(resp.json())
